from collections import Counter
import matplotlib.pyplot as plt
import matplotlib
import jieba
from pyecharts.charts import Pie, Bar, Grid, Liquid, Timeline, Page, Map, MapGlobe
from pyecharts.charts import WordCloud as W
from pyecharts import options as opts
from pyecharts.globals import ThemeType
from wordcloud import WordCloud
from PIL import Image
import numpy as np

matplotlib.rcParams['font.sans-serif'] = ['SimHei']  # 用黑体显示中文
matplotlib.rcParams['axes.unicode_minus'] = False  # 正常显示负号


def critics_number_bar(concat_pd):
    comment_time_list = concat_pd['评论时间'].tolist()
    ymd_list = [(date.split("-")[0] + '-' + date.split("-")[1]) for date in comment_time_list]
    temp_list = sorted(dict(Counter(ymd_list)).items(), key=lambda key: key[0])  # 按时间排序
    date_list, num_list = [], []
    for tpl in temp_list:
        date_list.append(tpl[0])
        num_list.append(tpl[1])
    limit = 36
    if len(date_list) > limit:  # 如果要呈现的数据过多则限制输出
        max_index = np.argmax(num_list)  # 筛选最大峰值前后的数据输出

    c = (
        Bar()
            .add_xaxis(date_list[max_index - 18:max_index + 18])
            .add_yaxis("评价条数", num_list[max_index - 18:max_index + 18])
            .set_global_opts(
            title_opts=opts.TitleOpts(title="不同时间影评人数"),
            datazoom_opts=opts.DataZoomOpts(),
            visualmap_opts=opts.VisualMapOpts(is_show=False, max_=90, min_=20, dimension=0),
            xaxis_opts=opts.AxisOpts(is_show=False)
        )
            .reversal_axis()
    )
    grid = Grid(init_opts=opts.InitOpts(height='1000px'))
    grid.add(c, grid_opts=opts.GridOpts(pos_right='10%', pos_left='30%'))
    return grid

def recommend_pie(concat_data):
    # 绘制饼状图
    temp_list = concat_data['观众推荐程度'].tolist()
    favor_list = [temp for temp in temp_list]
    favor_cnt = Counter(favor_list)
    favor_lst = sorted(dict(favor_cnt).items(), key=lambda key: key[1])
    label_list, num_list = [], []
    for tpl in favor_lst:
        label_list.append(tpl[0])
        num_list.append(tpl[1])
    data_pair = [list(z) for z in zip(label_list, num_list)]
    data_pair.sort(key=lambda x: x[1])

    pie = (
        Pie(init_opts=opts.InitOpts(width="1600px", height="800px", bg_color="#2c343c"))
            .add("",data_pair=data_pair, rosetype="radius", radius="55%", center=["50%", "50%"],
                label_opts=opts.LabelOpts(is_show=False, position="center"))
            .set_global_opts(title_opts=opts.TitleOpts(title="影评推荐指数比例", pos_left="left"),
                             legend_opts=opts.LegendOpts(type_="scroll", pos_left="90%", orient="vertical"))
            .set_series_opts(
            tooltip_opts=opts.TooltipOpts(
                trigger="item", formatter="{a} <br/>{b}: {c} ({d}%)"
            ),
            label_opts=opts.LabelOpts(color="rgba(255, 255, 255, 0.3)"),
        )
    )
    return pie

def comment_word_cloud1(concat_data):
    comments = concat_data['短评内容'].tolist()
    text = '\r'.join(comments)
    # 分词，返回结果为词的列表
    text_cut = jieba.lcut(text)
    # 将分好的词用某个符号分割开连成字符串
    text_cut = ' '.join(text_cut)
    # 导入停词
    stop_words = set("")
    stop_words.update(open("./dataset/stopwords.txt", encoding="utf-8").read().split("\n"))
    background = Image.open("./dataset/movie.jfif")
    graph = np.array(background)
    #new_list = [elem for elem in text_cut if elem not in stop_words]
    # 使用WordCloud生成词云
    word_cloud = WordCloud(font_path="simsun.ttc",
                           background_color="white",  # 词云图的背景颜色
                           mask=graph,
                           scale=8,
                           stopwords=stop_words)  # 去掉的停词
    word_cloud.generate(text_cut)
    # 运用matplotlib展现结果
    plt.subplots(figsize=(15, 10))
    plt.imshow(word_cloud, interpolation='bilinear')
    plt.axis("off")
    plt.savefig('_词云.png')
    # plt.show()
    plt.clf()
    print('_词云.png' + '绘制完毕')

def comment_word_cloud2(genre_list):
    a = (
        W().add("", genre_list, word_size_range=[20, 100], textstyle_opts=opts.TextStyleOpts(font_family="cursive"),)
            .set_global_opts(title_opts=opts.TitleOpts(title="WordCloud-自定义文字样式"))
    )
    return a

def movie_type_pie(genre_list):
    pie = (
        Pie().add("", genre_list, radius=["40%", "80%"], rosetype="radius")
            .set_global_opts(title_opts=opts.TitleOpts(title="电影类型分布", pos_left="left"),
                             legend_opts=opts.LegendOpts(type_="scroll", pos_left="90%", orient="vertical"))
            .set_series_opts(label_opts=opts.LabelOpts(formatter="{b}: {d}%"))
    )
    return pie

def all_movie_bar(sort_data):
    bar = (Bar().add_xaxis(sort_data['电影名称'].tolist()[::-1])
           .add_yaxis("", sort_data['评分'].tolist()[::-1], label_opts=opts.LabelOpts(position="right"))
           .set_global_opts(
        title_opts=opts.TitleOpts(title="电影评分排行榜单"),
        visualmap_opts=opts.VisualMapOpts(is_show=False, max_=100, min_=60),
        datazoom_opts=opts.DataZoomOpts(range_start=50, range_end=100, orient='vertical'),
        xaxis_opts=opts.AxisOpts(is_show=False))
           .reversal_axis()
           )
    grid = Grid(init_opts=opts.InitOpts(height='1000px'))
    grid.add(bar, grid_opts=opts.GridOpts(pos_right='10%', pos_left='30%'))
    return grid

def top20_movie_bar(sort_data):
    bar_ = Bar()
    bar_.add_xaxis(sort_data['电影名称'].tolist()[:20][::-1])
    bar_.add_yaxis("", sort_data['评分'].tolist()[:20][::-1],
                   label_opts=opts.LabelOpts(position="right"))
    bar_.set_global_opts(
        title_opts=opts.TitleOpts(title="Top20电影高评分榜"),
        visualmap_opts=opts.VisualMapOpts(is_show=False, max_=90, min_=20, dimension=0),
        xaxis_opts=opts.AxisOpts(is_show=False)
    )
    bar_.reversal_axis()
    grid_ = Grid(init_opts=opts.InitOpts(height='900px'))
    grid_.add(bar_, grid_opts=opts.GridOpts(pos_right='10%', pos_left='20%'))
    return grid_

def completion_liquid(sort_data, total):
    liq = (
        Liquid().add("lq", [len(sort_data) / total])
            .set_global_opts(title_opts=opts.TitleOpts(title="电影分析完成度"))
    )
    return liq

def movie_timeline(time_line, time_dict):
    tl = Timeline(init_opts=opts.InitOpts(theme=ThemeType.WONDERLAND))
    for times in time_line:
        if times == 'Nan':
            continue
        bar = (
            Bar(init_opts=opts.InitOpts(theme=ThemeType.WONDERLAND))
                .add_xaxis(time_dict[times]['movie'])
                .add_yaxis("电影评分", time_dict[times]['score'], label_opts=opts.LabelOpts(position="right"))
                .reversal_axis()
                .set_global_opts(
                title_opts=opts.TitleOpts("电影时间线")
            )
        )
        tl.add(bar, "{}".format(times))
    return tl

def map_chart(country_list) -> MapGlobe:
    name_map = {
    'Singapore Rep.': '新加坡',
    'Dominican Rep.': '多米尼加',
    'Palestine': '巴勒斯坦',
    'Bahamas': '巴哈马',
    'Timor-Leste': '东帝汶',
    'Afghanistan': '阿富汗',
    'Guinea-Bissau': '几内亚比绍',
    "Côte d'Ivoire": '科特迪瓦',
    'Siachen Glacier': '锡亚琴冰川',
    "Br. Indian Ocean Ter.": '英属印度洋领土',
    'Angola': '安哥拉',
    'Albania': '阿尔巴尼亚',
    'United Arab Emirates': '阿联酋',
    'Argentina': '阿根廷',
    'Armenia': '亚美尼亚',
    'French Southern and Antarctic Lands': '法属南半球和南极领地',
    'Australia': '澳大利亚',
    'Austria': '奥地利',
    'Azerbaijan': '阿塞拜疆',
    'Burundi': '布隆迪',
    'Belgium': '比利时',
    'Benin': '贝宁',
    'Burkina Faso': '布基纳法索',
    'Bangladesh': '孟加拉国',
    'Bulgaria': '保加利亚',
    'The Bahamas': '巴哈马',
    'Bosnia and Herz.': '波斯尼亚和黑塞哥维那',
    'Belarus': '白俄罗斯',
    'Belize': '伯利兹',
    'Bermuda': '百慕大',
    'Bolivia': '玻利维亚',
    'Brazil': '巴西',
    'Brunei': '文莱',
    'Bhutan': '不丹',
    'Botswana': '博茨瓦纳',
    'Central African Rep.': '中非',
    'Canada': '加拿大',
    'Switzerland': '瑞士',
    'Chile': '智利',
    'China': '中国',
    'Ivory Coast': '象牙海岸',
    'Cameroon': '喀麦隆',
    'Dem. Rep. Congo': '刚果民主共和国',
    'Congo': '刚果',
    'Colombia': '哥伦比亚',
    'Costa Rica': '哥斯达黎加',
    'Cuba': '古巴',
    'N. Cyprus': '北塞浦路斯',
    'Cyprus': '塞浦路斯',
    'Czech Rep.': '捷克',
    'Germany': '德国',
    'Djibouti': '吉布提',
    'Denmark': '丹麦',
    'Algeria': '阿尔及利亚',
    'Ecuador': '厄瓜多尔',
    'Egypt': '埃及',
    'Eritrea': '厄立特里亚',
    'Spain': '西班牙',
    'Estonia': '爱沙尼亚',
    'Ethiopia': '埃塞俄比亚',
    'Finland': '芬兰',
    'Fiji': '斐',
    'Falkland Islands': '福克兰群岛',
    'France': '法国',
    'Gabon': '加蓬',
    'United Kingdom': '英国',
    'Georgia': '格鲁吉亚',
    'Ghana': '加纳',
    'Guinea': '几内亚',
    'Gambia': '冈比亚',
    'Guinea Bissau': '几内亚比绍',
    'Eq. Guinea': '赤道几内亚',
    'Greece': '希腊',
    'Greenland': '格陵兰',
    'Guatemala': '危地马拉',
    'French Guiana': '法属圭亚那',
    'Guyana': '圭亚那',
    'Honduras': '洪都拉斯',
    'Croatia': '克罗地亚',
    'Haiti': '海地',
    'Hungary': '匈牙利',
    'Indonesia': '印度尼西亚',
    'India': '印度',
    'Ireland': '爱尔兰',
    'Iran': '伊朗',
    'Iraq': '伊拉克',
    'Iceland': '冰岛',
    'Israel': '以色列',
    'Italy': '意大利',
    'Jamaica': '牙买加',
    'Jordan': '约旦',
    'Japan': '日本',
    'Kazakhstan': '哈萨克斯坦',
    'Kenya': '肯尼亚',
    'Kyrgyzstan': '吉尔吉斯斯坦',
    'Cambodia': '柬埔寨',
    'Korea': '韩国',
    'Kosovo': '科索沃',
    'Kuwait': '科威特',
    'Lao PDR': '老挝',
    'Lebanon': '黎巴嫩',
    'Liberia': '利比里亚',
    'Libya': '利比亚',
    'Sri Lanka': '斯里兰卡',
    'Lesotho': '莱索托',
    'Lithuania': '立陶宛',
    'Luxembourg': '卢森堡',
    'Latvia': '拉脱维亚',
    'Morocco': '摩洛哥',
    'Moldova': '摩尔多瓦',
    'Madagascar': '马达加斯加',
    'Mexico': '墨西哥',
    'Macedonia': '马其顿',
    'Mali': '马里',
    'Myanmar': '缅甸',
    'Montenegro': '黑山',
    'Mongolia': '蒙古',
    'Mozambique': '莫桑比克',
    'Mauritania': '毛里塔尼亚',
    'Malawi': '马拉维',
    'Malaysia': '马来西亚',
    'Namibia': '纳米比亚',
    'New Caledonia': '新喀里多尼亚',
    'Niger': '尼日尔',
    'Nigeria': '尼日利亚',
    'Nicaragua': '尼加拉瓜',
    'Netherlands': '荷兰',
    'Norway': '挪威',
    'Nepal': '尼泊尔',
    'New Zealand': '新西兰',
    'Oman': '阿曼',
    'Pakistan': '巴基斯坦',
    'Panama': '巴拿马',
    'Peru': '秘鲁',
    'Philippines': '菲律宾',
    'Papua New Guinea': '巴布亚新几内亚',
    'Poland': '波兰',
    'Puerto Rico': '波多黎各',
    'Dem. Rep. Korea': '朝鲜',
    'Portugal': '葡萄牙',
    'Paraguay': '巴拉圭',
    'Qatar': '卡塔尔',
    'Romania': '罗马尼亚',
    'Russia': '俄罗斯',
    'Rwanda': '卢旺达',
    'W. Sahara': '西撒哈拉',
    'Saudi Arabia': '沙特阿拉伯',
    'Sudan': '苏丹',
    'S. Sudan': '南苏丹',
    'Senegal': '塞内加尔',
    'Solomon Is.': '所罗门群岛',
    'Sierra Leone': '塞拉利昂',
    'El Salvador': '萨尔瓦多',
    'Somaliland': '索马里兰',
    'Somalia': '索马里',
    'Serbia': '塞尔维亚',
    'Suriname': '苏里南',
    'Slovakia': '斯洛伐克',
    'Slovenia': '斯洛文尼亚',
    'Sweden': '瑞典',
    'Swaziland': '斯威士兰',
    'Syria': '叙利亚',
    'Chad': '乍得',
    'Togo': '多哥',
    'Thailand': '泰国',
    'Tajikistan': '塔吉克斯坦',
    'Turkmenistan': '土库曼斯坦',
    'East Timor': '东帝汶',
    'Trinidad and Tobago': '特里尼达和多巴哥',
    'Tunisia': '突尼斯',
    'Turkey': '土耳其',
    'Tanzania': '坦桑尼亚',
    'Uganda': '乌干达',
    'Ukraine': '乌克兰',
    'Uruguay': '乌拉圭',
    'United States': '美国',
    'Uzbekistan': '乌兹别克斯坦',
    'Venezuela': '委内瑞拉',
    'Vietnam': '越南',
    'Vanuatu': '瓦努阿图',
    'West Bank': '西岸',
    'Yemen': '也门',
    'South Africa': '南非',
    'Zambia': '赞比亚',
    'Zimbabwe': '津巴布韦',
    'Comoros': '科摩罗'
    }
    # re = requests.get(url).json()
    world = (
        Map()
        .add(series_name="电影数量", data_pair=country_list, maptype="world",
            is_map_symbol_show=False, name_map=name_map)
        .set_global_opts(
            title_opts=opts.TitleOpts(title='World Map'),
            visualmap_opts=opts.VisualMapOpts(
                max_=20,
                min_=1,
                is_piecewise=True)  # 定义图例为分段型，默认为连续的图例
        )
        .set_series_opts(label_opts=opts.LabelOpts(is_show=False))
    )
    return world

def dashboard1_page(concat_pd):
    page = Page(layout=Page.SimplePageLayout)
    comment_word_cloud1(concat_pd)
    page.add(
        critics_number_bar(concat_pd),
        recommend_pie(concat_pd)
    )
    page.render('dashboard1.html')

def dashboard2_page(genre_list, country_list, sort_data, time_line,time_dict, total):
    """
    :param genre_list: 电影剧情类型列表
    :param country_list: 制片国家列表
    :param sort_data: 按评分降序排序的pd数据
    :param time_line: 升序时间线
    :param time_dict: 时间线数据字典
    :param total: 总待爬取电影信息数
    :return: None
    """
    # page = Page(layout=Page.DraggablePageLayout)
    page = Page(layout=Page.SimplePageLayout)

    page.add(
        movie_type_pie(genre_list),
        top20_movie_bar(sort_data),
        completion_liquid(sort_data, total),
        all_movie_bar(sort_data),
        movie_timeline(time_line, time_dict),
        comment_word_cloud2(genre_list),
        map_chart(country_list)
    )
    page.render('dashboard2.html')
